import os
from graph.neo4j_client import Neo4jClient
from llm.ollama import OllamaClient
from response.answer_generator import AnswerGenerator

# 创建Neo4j和Ollama客户端实例
neo4j_client = Neo4jClient(
    uri=os.getenv("NEO4J_URI"),
    user=os.getenv("NEO4J_USER"),
    password=os.getenv("NEO4J_PASSWORD"),
    database=os.getenv("NEO4J_DATABASE")
)

ollama_client = OllamaClient(
    base_url=os.getenv("OLLAMA_API_BASE"),
    model=os.getenv("OLLAMA_MODEL")
)

# 创建AnswerGenerator实例
answer_generator = AnswerGenerator(ollama_client)

# 依赖注入
def get_neo4j_client():
    return neo4j_client

def get_ollama_client():
    return ollama_client

def get_answer_generator():
    return answer_generator